Quadratic Equation in Python